package com.spzx.common.rabbit.config;

import com.alibaba.fastjson2.JSON;
import com.spzx.common.rabbit.entity.GuiguCorrelationData;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;

/**
 * 服务器启动时，执行rabbitTemplate初始化，设置确认函数和回退函数
 * ApplicationEvent      一些子事件的父类。
 * ApplicationReadyEvent 具体子事件。表示应用程序启动好，IOC容器初始化好，存在相关bean对象了。再进行相关的初始化。
 * 也可以使用相关的注解替代： @EventListener
 */
@Slf4j
@Component
public class RabbitInitConfigApplicationListener implements ApplicationListener<ApplicationReadyEvent> {

    @Autowired
    RabbitTemplate rabbitTemplate;

    @Autowired
    RedisTemplate redisTemplate;


    @Override
    public void onApplicationEvent(ApplicationReadyEvent event) {

        this.setupCallbacks();
    }

    private void setupCallbacks(){
        /**
         * 只确认消息是否正确到达 Exchange 中,成功与否都会回调
         * @param correlationData 相关数据  非消息本身业务数据
         * @param ack             应答结果
         * @param reason           如果发送消息到交换器失败，错误原因
         */
        this.rabbitTemplate.setConfirmCallback(((correlationData,ack,reason) ->{
            log.info("correlationData={}",correlationData);
            if (ack){
                log.info("消息发送到Exchange成功：{}",correlationData);
            }else {
                log.error("消息发送到Exchange失败：{}",reason);
                try {
                    Thread.sleep(100);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                //消息重试,交换机没收到
                this.retrySendMsg(correlationData);
            }
        } ));
        /**
         * 消息没有正确到达队列时触发回调，如果正确到达队列不执行
         */
        this.rabbitTemplate.setReturnsCallback(returnedMessage -> {
            log.error("Returned: " + returnedMessage.getMessage() + "\nreplyCode: " + returnedMessage.getReplyCode()
                    + "\nreplyText: " + returnedMessage.getReplyText() + "\nexchange/rk: "
                    + returnedMessage.getExchange() + "/" + returnedMessage.getRoutingKey());
            //路由失败也要重发
            String uuid=returnedMessage.getMessage().getMessageProperties().
                    getHeader("spring_returned_message_correlation");
//            GuiguCorrelationData guiguCorrelationData = (GuiguCorrelationData) redisTemplate.opsForValue().get(uuid);
            String json = (String) redisTemplate.opsForValue().get(uuid);
            if (json==null) {
                log.info("redis中未找到correlationData，uuid：{}",uuid);
            }
            GuiguCorrelationData guiguCorrelationData = JSON.parseObject(json, GuiguCorrelationData.class);
            if (guiguCorrelationData!=null && !guiguCorrelationData.isDelay()) {
                try {
                    Thread.sleep(100);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                retrySendMsg(guiguCorrelationData);
            }
        });
    }

    private void retrySendMsg(CorrelationData correlationData) {
        GuiguCorrelationData guiguCorrelationData=(GuiguCorrelationData) correlationData;
        int retryCount = guiguCorrelationData.getRetryCount();
        if (retryCount>=3) {
            log.info("消息到达最大重试次数");
            return;
        }
        retryCount ++;
        guiguCorrelationData.setRetryCount(retryCount);
        redisTemplate.opsForValue().set(guiguCorrelationData.getId(),
                JSON.toJSONString(guiguCorrelationData),10, TimeUnit.MINUTES);
        rabbitTemplate.convertAndSend(guiguCorrelationData.getExchange(),
                guiguCorrelationData.getRoutingKey(),guiguCorrelationData.getMessage(),guiguCorrelationData);
        log.info("进行消息重发");
    }


}
